# Copyright (c) 2018, Hewlett Packard Enterprise Development LP
# GNU General Public License v3.0+ (see LICENSES/GPL-3.0-or-later.txt or https://www.gnu.org/licenses/gpl-3.0.txt)
# SPDX-License-Identifier: GPL-3.0-or-later

from __future__ import annotations

import sys
from unittest import mock

sys.modules["hpe3par_sdk"] = mock.Mock()
sys.modules["hpe3par_sdk.client"] = mock.Mock()
sys.modules["hpe3parclient"] = mock.Mock()
sys.modules["hpe3parclient.exceptions"] = mock.Mock()
from ansible_collections.community.general.plugins.modules import ss_3par_cpg
from ansible_collections.community.general.plugins.module_utils.storage.hpe3par import hpe3par


@mock.patch("ansible_collections.community.general.plugins.modules.ss_3par_cpg.client")
@mock.patch("ansible_collections.community.general.plugins.modules.ss_3par_cpg.AnsibleModule")
@mock.patch("ansible_collections.community.general.plugins.modules.ss_3par_cpg.create_cpg")
def test_module_args(mock_create_cpg, mock_module, mock_client):
    """
    hpe3par CPG - test module arguments
    """

    PARAMS_FOR_PRESENT = {
        "storage_system_ip": "192.168.0.1",
        "storage_system_username": "USER",
        "storage_system_password": "PASS",
        "cpg_name": "test_cpg",
        "domain": "test_domain",
        "growth_increment": 32768,
        "growth_increment_unit": "MiB",
        "growth_limit": 32768,
        "growth_limit_unit": "MiB",
        "growth_warning": 32768,
        "growth_warning_unit": "MiB",
        "raid_type": "R6",
        "set_size": 8,
        "high_availability": "MAG",
        "disk_type": "FC",
        "state": "present",
        "secure": False,
    }
    mock_module.params = PARAMS_FOR_PRESENT
    mock_module.return_value = mock_module
    mock_client.HPE3ParClient.login.return_value = True
    mock_create_cpg.return_value = (True, True, "Created CPG successfully.")
    ss_3par_cpg.main()
    mock_module.assert_called_with(
        argument_spec=hpe3par.cpg_argument_spec(), required_together=[["raid_type", "set_size"]]
    )


@mock.patch("ansible_collections.community.general.plugins.modules.ss_3par_cpg.client")
@mock.patch("ansible_collections.community.general.plugins.modules.ss_3par_cpg.AnsibleModule")
@mock.patch("ansible_collections.community.general.plugins.modules.ss_3par_cpg.create_cpg")
def test_main_exit_functionality_present_success_without_issue_attr_dict(mock_create_cpg, mock_module, mock_client):
    """
    hpe3par flash cache - success check
    """
    PARAMS_FOR_PRESENT = {
        "storage_system_ip": "192.168.0.1",
        "storage_system_name": "3PAR",
        "storage_system_username": "USER",
        "storage_system_password": "PASS",
        "cpg_name": "test_cpg",
        "domain": "test_domain",
        "growth_increment": 32768,
        "growth_increment_unit": "MiB",
        "growth_limit": 32768,
        "growth_limit_unit": "MiB",
        "growth_warning": 32768,
        "growth_warning_unit": "MiB",
        "raid_type": "R6",
        "set_size": 8,
        "high_availability": "MAG",
        "disk_type": "FC",
        "state": "present",
        "secure": False,
    }
    # This creates a instance of the AnsibleModule mock.
    mock_module.params = PARAMS_FOR_PRESENT
    mock_module.return_value = mock_module
    instance = mock_module.return_value
    mock_client.HPE3ParClient.login.return_value = True
    mock_create_cpg.return_value = (True, True, "Created CPG successfully.")
    ss_3par_cpg.main()
    # AnsibleModule.exit_json should be called
    instance.exit_json.assert_called_with(changed=True, msg="Created CPG successfully.")
    # AnsibleModule.fail_json should not be called
    assert instance.fail_json.call_count == 0


@mock.patch("ansible_collections.community.general.plugins.modules.ss_3par_cpg.client")
@mock.patch("ansible_collections.community.general.plugins.modules.ss_3par_cpg.AnsibleModule")
@mock.patch("ansible_collections.community.general.plugins.modules.ss_3par_cpg.delete_cpg")
def test_main_exit_functionality_absent_success_without_issue_attr_dict(mock_delete_cpg, mock_module, mock_client):
    """
    hpe3par flash cache - success check
    """
    PARAMS_FOR_DELETE = {
        "storage_system_ip": "192.168.0.1",
        "storage_system_name": "3PAR",
        "storage_system_username": "USER",
        "storage_system_password": "PASS",
        "cpg_name": "test_cpg",
        "domain": None,
        "growth_increment": None,
        "growth_increment_unit": None,
        "growth_limit": None,
        "growth_limit_unit": None,
        "growth_warning": None,
        "growth_warning_unit": None,
        "raid_type": None,
        "set_size": None,
        "high_availability": None,
        "disk_type": None,
        "state": "absent",
        "secure": False,
    }
    # This creates a instance of the AnsibleModule mock.
    mock_module.params = PARAMS_FOR_DELETE
    mock_module.return_value = mock_module
    instance = mock_module.return_value
    mock_delete_cpg.return_value = (True, True, "Deleted CPG test_cpg successfully.")
    mock_client.HPE3ParClient.login.return_value = True
    ss_3par_cpg.main()
    # AnsibleModule.exit_json should be called
    instance.exit_json.assert_called_with(changed=True, msg="Deleted CPG test_cpg successfully.")
    # AnsibleModule.fail_json should not be called
    assert instance.fail_json.call_count == 0


def test_convert_to_binary_multiple():
    assert hpe3par.convert_to_binary_multiple(None) == -1
    assert hpe3par.convert_to_binary_multiple("-1.0 MiB") == -1
    assert hpe3par.convert_to_binary_multiple("-1.0GiB") == -1
    assert hpe3par.convert_to_binary_multiple("1.0   MiB") == 1
    assert hpe3par.convert_to_binary_multiple("1.5GiB") == 1.5 * 1024
    assert hpe3par.convert_to_binary_multiple("1.5 TiB") == 1.5 * 1024 * 1024
    assert hpe3par.convert_to_binary_multiple(" 1.5 TiB ") == 1.5 * 1024 * 1024


@mock.patch("ansible_collections.community.general.plugins.modules.ss_3par_cpg.client")
def test_validate_set_size(mock_client):
    mock_client.HPE3ParClient.RAID_MAP = {
        "R0": {"raid_value": 1, "set_sizes": [1]},
        "R1": {"raid_value": 2, "set_sizes": [2, 3, 4]},
        "R5": {"raid_value": 3, "set_sizes": [3, 4, 5, 6, 7, 8, 9]},
        "R6": {"raid_value": 4, "set_sizes": [6, 8, 10, 12, 16]},
    }
    raid_type = "R0"
    set_size = 1
    assert ss_3par_cpg.validate_set_size(raid_type, set_size)

    set_size = 2
    assert not ss_3par_cpg.validate_set_size(raid_type, set_size)

    raid_type = None
    assert not ss_3par_cpg.validate_set_size(raid_type, set_size)


@mock.patch("ansible_collections.community.general.plugins.modules.ss_3par_cpg.client")
def test_cpg_ldlayout_map(mock_client):
    mock_client.HPE3ParClient.PORT = 1
    mock_client.HPE3ParClient.RAID_MAP = {
        "R0": {"raid_value": 1, "set_sizes": [1]},
        "R1": {"raid_value": 2, "set_sizes": [2, 3, 4]},
        "R5": {"raid_value": 3, "set_sizes": [3, 4, 5, 6, 7, 8, 9]},
        "R6": {"raid_value": 4, "set_sizes": [6, 8, 10, 12, 16]},
    }
    ldlayout_dict = {"RAIDType": "R6", "HA": "PORT"}
    assert ss_3par_cpg.cpg_ldlayout_map(ldlayout_dict) == {"RAIDType": 4, "HA": 1}


@mock.patch("ansible_collections.community.general.plugins.modules.ss_3par_cpg.client")
def test_create_cpg(mock_client):
    ss_3par_cpg.validate_set_size = mock.Mock(return_value=True)
    ss_3par_cpg.cpg_ldlayout_map = mock.Mock(return_value={"RAIDType": 4, "HA": 1})

    mock_client.HPE3ParClient.login.return_value = True
    mock_client.HPE3ParClient.cpgExists.return_value = False
    mock_client.HPE3ParClient.FC = 1
    mock_client.HPE3ParClient.createCPG.return_value = True

    assert ss_3par_cpg.create_cpg(
        mock_client.HPE3ParClient,
        "test_cpg",
        "test_domain",
        "32768 MiB",
        "32768 MiB",
        "32768 MiB",
        "R6",
        8,
        "MAG",
        "FC",
    ) == (True, True, "Created CPG test_cpg successfully.")

    mock_client.HPE3ParClient.cpgExists.return_value = True
    assert ss_3par_cpg.create_cpg(
        mock_client.HPE3ParClient,
        "test_cpg",
        "test_domain",
        "32768.0 MiB",
        "32768.0 MiB",
        "32768.0 MiB",
        "R6",
        8,
        "MAG",
        "FC",
    ) == (True, False, "CPG already present")

    ss_3par_cpg.validate_set_size = mock.Mock(return_value=False)
    assert ss_3par_cpg.create_cpg(
        mock_client.HPE3ParClient,
        "test_cpg",
        "test_domain",
        "32768.0 MiB",
        "32768 MiB",
        "32768.0 MiB",
        "R6",
        3,
        "MAG",
        "FC",
    ) == (False, False, "Set size 3 not part of RAID set R6")


@mock.patch("ansible_collections.community.general.plugins.modules.ss_3par_cpg.client")
def test_delete_cpg(mock_client):
    mock_client.HPE3ParClient.login.return_value = True
    mock_client.HPE3ParClient.cpgExists.return_value = True
    mock_client.HPE3ParClient.FC = 1
    mock_client.HPE3ParClient.deleteCPG.return_value = True

    assert ss_3par_cpg.delete_cpg(mock_client.HPE3ParClient, "test_cpg") == (
        True,
        True,
        "Deleted CPG test_cpg successfully.",
    )

    mock_client.HPE3ParClient.cpgExists.return_value = False

    assert ss_3par_cpg.delete_cpg(mock_client.HPE3ParClient, "test_cpg") == (True, False, "CPG does not exist")
    assert ss_3par_cpg.delete_cpg(mock_client.HPE3ParClient, None) == (True, False, "CPG does not exist")
